package me.zhengjie.utils;

import cn.hutool.extra.qrcode.QrCodeUtil;
import cn.hutool.extra.qrcode.QrConfig;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;

public class QRCodeGeneration {
    /**
     * 生成中间带logo的二维码
     * @param content  二维码内容
     * @param logoPath logo图标的地址
     * @return BufferedImage 二维码图片
     */
    public  BufferedImage GenerateQrCodeLogo(String content, String logoPath) {
        BufferedImage image = QrCodeUtil.generate(
                content, // 二维码内容
                QrConfig.create().setImg(logoPath)// 附带logo
                        .setWidth(290) // 二维码的宽
                        .setHeight(290) // 二维码的高
                        .setMargin(0)); // 边距
        return image;
    }
    /**
     * 生成中间带logo的二维码
     * @param content  二维码内容
     * @param widthHeight  二维码宽高
     * @param margin  二维码内边距
     * @return BufferedImage 二维码图片
     */
    public  BufferedImage GenerateQrCode(String content,Integer widthHeight,Integer margin) {
        BufferedImage image = QrCodeUtil.generate(
                content, // 二维码内容
                QrConfig.create()
                        .setWidth(widthHeight) // 二维码的宽
                        .setHeight(widthHeight) // 二维码的高
                        .setMargin(margin)); // 边距
        return image;
    }

    /**
     * 将二维码写入背景中
     *
     * @param codeFile  二维码
     * @param bgFile   背景图片的路径
     * @throws IOException
     * @return BufferedImage 带背景的二维码图片
     */
    public  BufferedImage GenerateQrCodeBackground(File codeFile, File bgFile) throws IOException {
            //画背景图片
            BufferedImage bgImage = ImageIO.read(bgFile);
            //二维码
            BufferedImage codeImage = ImageIO.read(codeFile);
            // 背景图片的宽度
            int bgWidth = bgImage.getWidth();
            // 二维码的宽度
            int qrWidth = codeImage.getWidth();
            // 二维码距离背景图片横坐标（X）的距离，居中显示
            int distanceX = (bgWidth - qrWidth) / 2;
            // 二维码距离背景图片纵坐标（Y）的距离
            int distanceY = distanceX;
            // 基于图片backgroundImage对象打开绘图
            Graphics2D g = bgImage.createGraphics();
            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP));
            g.drawImage(codeImage, distanceX, distanceY, null);
            g.dispose();
            return bgImage;
    }
}

